leetcode 2272 solution in python